import unittest
import os.path
import time
# 导入HTMLTestRunner扩展
from HTMLTestRunner import HTMLTestRunner

# loader
loader = unittest.defaultTestLoader.discover('test_cases', 'test*')

# runner
# 准备要写入的文件
# 文件名规则: 年月日时分秒-项目(模块名).html
report = os.path.abspath('report')
filename = time.strftime('%Y%m%d-%H%M%S') + '_demo.html'
filepath = os.path.join(report, filename)

# 打开文件
f = open(filepath, mode='wb')

# TODO 返回一个runner, 配置测试报告信息
runner = HTMLTestRunner(
    title='这里写测试报告的标题',
    description='这里写测试报告的描述信息',
    tester='tester01',
    stream=f,
    verbosity=2
)
runner.run(loader)

# 关闭文件
f.close()
